package com.dexter.year2023.charpter6_tree_level_travel.level2.topic2_1层次遍历;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * 429. N 叉树的层序遍历
 * https://leetcode.cn/problems/n-ary-tree-level-order-traversal/
 * <p>
 * 给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。
 * 树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。
 * <p>
 * 示例 1：
 * 输入：root = [1,null,3,2,4,null,5,6]
 * 输出：[[1],[3,2,4],[5,6]]
 * 示例 2：
 * 输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
 * 输出：[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
 * <p>
 * 提示：
 * 树的高度不会超过 1000
 * 树的节点总数在 [0, 10^4] 之间
 *
 * @author Dexter
 */
public class LC429NLevelOrder {
    public static void main(String[] args) {
        NTreeNode nTreeNode = NTreeNode.buildBinaryTree();
        // 提交结果：已通过 - 时间击败88.53% 的Java用户，内存击败42.14% 的Java用户
        System.out.println(nLevelOrder(nTreeNode));
    }

    public static List<List<Integer>> nLevelOrder(NTreeNode root) {
        // 结果 - 数组集合
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        // 队列 - 链表集合
        LinkedList<NTreeNode> q = new LinkedList<NTreeNode>();
        // 判空
        if (root != null) {
            q.add(root);
        }
        while (!q.isEmpty()) {
            // 下一行存储的队列 - 链表集合
            LinkedList<NTreeNode> next = new LinkedList<NTreeNode>();
            // 子集 - 数组集合
            List<Integer> temp = new ArrayList<Integer>();
            while (!q.isEmpty()) {
                NTreeNode cur = q.removeFirst();
                temp.add(cur.val);
                // 遍历N个孩子
                for (NTreeNode chd : cur.children) {
                    if (chd != null) {
                        next.add(chd);
                    }
                }
            }
            // 开始下一行的队列
            q = next;
            res.add(temp);
        }
        return res;
    }
}

/**
 * N叉树の节点定义
 */
class NTreeNode {
    public int val;
    public List<NTreeNode> children;

    public NTreeNode() {
    }

    public NTreeNode(int _val) {
        val = _val;
    }

    public NTreeNode(int _val, List<NTreeNode> _children) {
        val = _val;
        children = _children;
    }

    public static NTreeNode buildBinaryTree() {
        List<NTreeNode> list11 = new ArrayList<>();
        List<NTreeNode> list12 = new ArrayList<>();
        List<NTreeNode> list13 = new ArrayList<>();
        List<NTreeNode> list14 = new ArrayList<>();

        List<NTreeNode> list3 = new ArrayList<>();
        list3.add(new NTreeNode(5, list11));
        list3.add(new NTreeNode(6, list12));
        NTreeNode nTreeNode3 = new NTreeNode(3, list3);

        List<NTreeNode> list1 = new ArrayList<>();
        list1.add(nTreeNode3);
        list1.add(new NTreeNode(2, list13));
        list1.add(new NTreeNode(4, list14));

        NTreeNode nTreeNode = new NTreeNode(1, list1);
        return nTreeNode;
    }
}
